package com.atfcm.link;

import javax.crypto.interfaces.PBEKey;
import java.util.Arrays;
import java.util.Collection;
import java.util.LinkedList;
import java.util.Objects;

/**
 * @Author whaty
 * @create 2024/10/25 13:46
 */
class MyLinkedList {
    public ListNode head;

    public int size = 0;

    public static void main(String[] args) {
        MyLinkedList list = new MyLinkedList();
        list.addAtTail(1);
        list.addAtTail(2);
        list.addAtTail(3);
        list.addAtTail(4);

        printList(list.head);

    }

    public MyLinkedList() {

    }

    /**
     * 获取链表中下标为 index 的节点的值。如果下标无效，则返回 -1
     *
     * @param index
     * @return
     */
    public int get(int index) {
        if (index < 0 || head == null) {
            return -1;
        }

        ListNode cur = head;
        for (int i = 0; i < index; i++) {
            if (cur == null) {
                return -1;
            }
            cur = cur.next;
        }
        return cur == null ? -1 : cur.val;
    }

    /**
     * 将一个值为 val 的节点插入到链表中第一个元素之前。在插入完成后，新节点会成为链表的第一个节点
     *
     * @param val
     */
    public void addAtHead(int val) {
        if (head == null) {
            head = new ListNode(val);
        } else {
            ListNode newNode = new ListNode(val);
            newNode.next = head;
            head = newNode;
        }
        size++;
    }

    /**
     * 将一个值为 val 的节点追加到链表中作为链表的最后一个元素
     *
     * @param val
     */
    public void addAtTail(int val) {
        if (head == null) {
            head = new ListNode(val);
        } else {
            ListNode cur = head;
            while (cur.next != null) {
                cur = cur.next;
            }
            cur.next = new ListNode(val);
        }
        size++;
    }

    /**
     * //     * 将一个值为 val 的节点插入到链表中下标为 index 的节点之前。
     * 如果index等于链表的长度，那么该节点会被追加到链表的末尾。如果 index 比长度更大，该节点将 不会插入 到链表中。
     *
     * @param index
     * @param val
     */
    public void addAtIndex(int index, int val) {

        if (index < 0 || index > size) {
            return;
        }

        if (index == 0) {
            addAtHead(val);
            return;
        }
        if (index == size) {
            addAtTail(val);
            return;
        }

        ListNode cur = head;
        for (int i = 0; i < index - 1; i++) {
            cur = cur.next;
        }
        ListNode newNode = new ListNode(val);
        newNode.next = cur.next;
        cur.next = newNode;
        size++;


    }

    /**
     * 如果下标有效，则删除链表中下标为 index 的节点。
     *
     * @param index
     */
    public void deleteAtIndex(int index) {
        if(index < 0 || index >= size || head == null) {
            return;
        }

        if(index == 0) {
            head = head.next;
            //注意--别忘了写
            size-- ;
            return;
        }
        ListNode cur = head;
        for (int i = 0; i < index - 1; i++) {
            cur = cur.next;
        }

        cur.next = cur.next.next;
        size-- ;
    }

    public static void printList(ListNode listNode) {
        String val = "";
        if (listNode != null) {
            ListNode current = listNode;
            while (current != null) {
                val += current.val + " -> ";
                current = current.next;
            }
        }
        System.out.println(val);
    }

   static   class ListNode {
        public int val;
        public ListNode next;

        ListNode(int val) {
            this.val = val;
        }
    }

}

